classification using tensorflow